armstrong number in python program